#!/usr/bin/env python
# -*- coding:utf-8 -*-
from .models import *


class LLMFactory:
    MODEL_MAPPING = {
        'deepseek': DeepSeekModel,
        'zhipu': ZhiPuModel,
    }

    @classmethod
    def get_model(cls, model_name: str, api_key: str):
        return cls.MODEL_MAPPING[model_name](api_key).client
